from setuptools import Extension, find_packages, setup
from setuptools.dist import Distribution


# 定义扩展模块
example_module = Extension(
    'torch._C',
    sources=['src/main.c']
)

# 调用 setup 函数进行构建
setup(
    name='torch',
    version='1.0',
    description='An example Python C extension module',
    ext_modules=[example_module]
)